"""example of how to compute curvature of edges"""
import networkx as nx
import numpy as np

from geometric_clustering import compute_curvatures

if __name__ == "__main__":
    graph = nx.read_gpickle("graph_powergrid.gpickle")

    t_min = -2.0
    t_max = 1.0
    n_t = 50
    times = np.logspace(t_min, t_max, n_t)

    kappas = compute_curvatures(graph, times, n_workers=10, filename="curvature.pkl")
